#pragma once

#include <unistd.h>
#include <fcntl.h>
#include "Log.hpp"

// 设置为非阻塞模式
void SetNonBlockOrDie(int sock)
{
    int fl = fcntl(sock, F_GETFL);
    if (fl < 0) 
    {
        lg(Error, "fcntl error!");
        exit(5);
    }
    fcntl(sock, F_SETFL, fl | O_NONBLOCK);
}